package com.kaesar.leetcode.LeetCode_051_100;

import java.util.ArrayList;
import java.util.List;

public class LeetCode_073 {
    public static void setZeroes(int[][] matrix) {
        List<Integer> zeroRows = new ArrayList<>();
        List<Integer> zeroColumns = new ArrayList<>();
        // 获取哪些行存在元素为0
        for (int i = 0; i < matrix.length; i++) {
            for (int j = 0; j < matrix[i].length; j++) {
                if (matrix[i][j] == 0) {
                    zeroRows.add(i);
                    break;
                }
            }
        }

        // 获取哪些列存在元素为0
        for (int i = 0; i < matrix[0].length; i++) {
            for (int j = 0; j < matrix.length; j++) {
                if (matrix[j][i] == 0) {
                    zeroColumns.add(i);
                    break;
                }
            }
        }

        // 将存在元素为0的行全都置为0
        for (int i = 0; i < matrix.length; i++) {
            if (zeroRows.contains(i)) {
                for (int j = 0; j < matrix[i].length; j++) {
                    matrix[i][j] = 0;
                }
            }
        }

        // 将存在元素为0的列全都置为0
        for (int j = 0; j < matrix[0].length; j++) {
            if (zeroColumns.contains(j)) {
                for (int i = 0; i < matrix.length; i++) {
                    matrix[i][j] = 0;
                }
            }
        }
    }

    public static void main(String[] args) {
        int[][] matrix = new int[][]{{1, 1, 1}, {1, 0, 1}, {1, 1, 1}};
        setZeroes(matrix);
        for (int[] ints : matrix) {
            for (int anInt : ints) {
                System.out.print(anInt);
            }
            System.out.println();
        }
    }
}
